/**
 * 平衡二叉树
 *
 * 给定一个二叉树，判断它是否是 平衡二叉树
 *
 * 示例 1：
 * 输入：root = [3,9,20,null,null,15,7]
 * 输出：true
 *
 * 示例 2：
 * 输入：root = [1,2,2,3,3,null,null,4,4]
 * 输出：false
 *
 * 示例 3：
 * 输入：root = []
 * 输出：true
 *
 * 提示：
 * 树中的节点数在范围 [0, 5000] 内
 * -104 <= Node.val <= 104
 */

/**
 * 这题我们用递归, 用递归, 首先递归
 * 函数帮我们判断这个根节点是否为平衡
 * 二叉树, 然后分别遍历他的左子树和右子树
 * 就可以了, 这其中判断是否为平衡二叉树的方法
 * 要求出两边的高度, 高度差不能 大于 1
 * 时间复杂度 : O(n ^ 2)
 * 空间复杂度 : O(n)
 */

public class Main {
    public boolean isBalanced(TreeNode root) {

        if (root == null) {
            return true;
        }

        int left = dfs(root.left);
        int right = dfs(root.right);

        // 两边高度差
        if (Math.abs(right - left) > 1) {
            return false;
        }

        return isBalanced(root.left) && isBalanced(root.right);

    }

    // 求 root围殴根节点 的最大高度
    public int dfs(TreeNode root) {
        if (root == null) {
            return 0;
        }

        int left = dfs(root.left) + 1;
        int right = dfs(root.right) + 1;

        return Math.max(left, right);
    }
}

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode() {}
    TreeNode(int val) { this.val = val; }
    TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }
}
